package com.xzz.stack;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Stack;

/**
 * @author: hhz
 * @create: 2021-12-29 14:54
 * 单调栈
 * 496. 下一个更大元素 I
 * 输入：nums1 = [4,1,2], nums2 = [1,3,4,2].
 * 输出：[-1,3,-1]
 * 解释：nums1 中每个值的下一个更大元素如下所述：
 * - 4 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
 * - 1 ，用加粗斜体标识，nums2 = [1,3,4,2]。下一个更大元素是 3 。
 * - 2 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
 **/
public class NextGreaterElement {
    public static void main(String[] args) {
        System.out.println(Arrays.toString(nextGreaterElement(new int[]{4,1,2},new int[]{1,3,4,2})));
        System.out.println(Arrays.toString(nextGreaterElement(new int[]{2,4},new int[]{1,2,3,4})));
    }
    public static int[] nextGreaterElement(int[] nums1, int[] nums2) {
        HashMap<Integer,Integer> map = new HashMap<>();
        int[] res = new int[nums1.length];
        for (int i = 0; i < nums1.length; i++) {
            map.put(nums1[i],i);
        }
        Stack<Integer> stack = new Stack<>();
        for (int i = nums2.length-1; i >=0; i--) {
            while (!stack.empty() && stack.peek()<=nums2[i]){
                stack.pop();
            }
            if(map.containsKey(nums2[i])){
                res[map.get(nums2[i])] = stack.empty()?-1:stack.peek();
            }
            stack.push(nums2[i]);
        }
        return res;
    }
}
